package com.liu.design.structure.flyweight;

import java.util.HashMap;

/**
 * @author LIU
 * @version 1.0
 * @date 2024/5/6 17:16
 * @description 根据需要返回压一个网站
 */
public class WebSiteFactory {
    //集合， 充当池的作用
    private HashMap<String,WebSite> pool = new HashMap<>();

    //根据网站的类型，返回一个网站, 如果没有就创建一个网站，并放入到池中,并返回
    public WebSite getWebSiteCategory( String type){
        if (!pool.containsKey(type)) {
            pool.put(type,new ConcreteWebSite(type));
        }
        return pool.get(type);
    }

    //获取网站分类的总数 (池中有多少个网站类型)
    public int getWebSiteCount() {
        return pool.size();
    }
}
